#ifndef CATEGORYMODEL_H
#define CATEGORYMODEL_H

#include <QAbstractListModel>
#include <QVector>

#include "keybasedlistmodel.h"

class CategoryCollection;
class GameCategory;

class CategoryModel : public KeyBasedListModel
{
    Q_OBJECT
public:
    explicit CategoryModel(const CategoryCollection* categories, QObject *parent = 0);

    virtual QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;

protected:
    QVariant getDataForKey(const QString& key, int column, int role) const;

private:
    const CategoryCollection *_categories;
};

#endif // CATEGORYMODEL_H
